package com.sfx.concurrent.demo.concurrent.ThreadPoll;

import lombok.extern.slf4j.Slf4j;

import java.util.HashSet;
import java.util.concurrent.LinkedBlockingDeque;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.LongAdder;

/**
 * 任务生产者生产的任务对象交给线程池
 * 如果没有线程就创建线程执行
 * 如果有线程但线程数量不足,就像任务放到阻塞队列中
 */

/**
 * 拒绝策略->定义一个方法,让调用者决定怎么执行->队列+任务
 * 函数式接口
 */
@FunctionalInterface
interface RejectPolicy<T>{
    void reject(MyBlockingQueue<T> queue, T task);
}
@Slf4j(topic = "c.MyThreadPool")
public class MyThreadPool {
    // 任务队列
    private MyBlockingQueue<Runnable> taskQueue;
    // 线程集合
    private HashSet<Worker> workers = new HashSet<>();
    // 核心线程数
    private int coreSize;
    // 获取任务的超时时间
    private long timeout;
    // 时间单位
    private TimeUnit unit;
    //拒绝策略
    private RejectPolicy<Runnable> rejectPolicy;

    public MyThreadPool(int coreSize, long timeout, TimeUnit unit,int queueCapacity,RejectPolicy<Runnable> rejectPolicy) {
        this.taskQueue = new MyBlockingQueue<>(queueCapacity);
        this.coreSize = coreSize;
        this.timeout = timeout;
        this.unit = unit;
        this.rejectPolicy = rejectPolicy;
    }

    //执行任务
    public void execute(Runnable task) {
        synchronized (workers) {
            if(workers.size()<this.coreSize) {
                // 如果任务数(当前线程数)<线程核心数 那么就交给工作线程来执行该任务
                Worker worker = new Worker(task);
                log.debug("新增worker{},任务{}",worker,task);
                workers.add(worker);
                worker.start();
            }else {
                //如果任务数(当前线程数)>线程核心数 那么就将任务放到任务队列中
                //log.debug("加入到任务队列{}",task);
                //taskQueue.add(task);//死等
                taskQueue.tryPut(rejectPolicy,task);
            }
        }
    }

    //工作线程来执行任务
    class Worker extends Thread{
        private Runnable task;
        public Worker(Runnable task) {
            this.task = task;
        }

        //该工作线程要执行这个task任务
        //如果task任务执行完了,继续去看任务队列里面还有没有任务
        @Override
        public void run() {
            while(this.task!=null || (task= taskQueue.poll(timeout,unit))!=null) {
                try{
                    log.debug("执行任务中{}",task);
                    task.run();
                }catch (Exception e) {
                    e.printStackTrace();
                }finally {
                    //该任务执行完毕之后,进行销毁
                    task = null;
                }
            }

            //该任务也执行完了,并且任务队列里面也没有任务了
            //证明该线程也执行完毕了,将当前线程移除线程集合中
            synchronized (workers) {
                log.debug("该线程执行完任务被移除{}",this);
                workers.remove(this);
            }
        }
    }
}